package com.spzx.report.aiTools;

import lombok.SneakyThrows;
import org.springframework.core.io.ClassPathResource;
import org.springframework.core.io.Resource;
import org.springframework.util.FileCopyUtils;

import java.io.InputStreamReader;

// 提示词生成器核心类
public class PromptGenerator {

    // 字符串提示词模板
    public static final String PROMPT_TEMPLATE = "";// 简单的提示词可以直接字符串

    // 文本提示词模板
    public static String systemPrompt = "你是一个精通电商领域的助手，需用根据用户的要求准确回答问题。" + PromptGenerator.readPromptFile();// 系统提示词信息

    public static String systemChatPrompt = "你是一个精通电商领域的助手，需用根据用户的要求准确回答问题。" + PromptGenerator.chatPromptFile();// 系统提示词信息



    @SneakyThrows
    public static String readPromptFile()  {
        Resource resource = new ClassPathResource("aiReportPrompt.txt");// 此处读取提示词文本
        String content = FileCopyUtils.copyToString(new InputStreamReader(resource.getInputStream()));
        return content;
    }

    @SneakyThrows
    public static String chatPromptFile()  {
        Resource resource = new ClassPathResource("aiChatPrompt.txt");// 此处读取提示词文本
        String content = FileCopyUtils.copyToString(new InputStreamReader(resource.getInputStream()));
        return content;
    }

}